package com.tianji.aigc.agent;

import com.tianji.aigc.config.ai.SystemPromptConfig;
import com.tianji.aigc.enums.AgentTypeEnum;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;

//路由智能体
@Component
@RequiredArgsConstructor
public class RouteAgent extends AbstractAgent{

    private final SystemPromptConfig systemPromptConfig;

    @Override
    public String systemMessage() {
        return systemPromptConfig.getRouteAgentSystemMessage().get();
    }

    //AbstractAgent 是抽象类，可以不实现接口方法。
    //RouteAgent 是具体类，必须实现 getAgentType() 方法
    @Override
    public AgentTypeEnum getAgentType() {
        return AgentTypeEnum.ROUTE;
    }

}
